#pragma onece

#include <iostream>
#include <queue>
#include <pthread.h>

template <class T>
class BlockQueue
{
    static const int default_num = 20; // 不能只写default，可能会重名
public:
    BlockQueue(int max_cap = default_num) : _max_cap(max_cap)
    {
        std::cout << "mutex and cond already" << std::endl;
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_c_cond, nullptr);
        pthread_cond_init(&_p_cond, nullptr);
        // _low_water = _max_cap/3;
        // _high_water = (_max_cap * 2) / 3;
    }

    T Pop()
    {
        pthread_mutex_lock(&_mutex);
        // if(_q.size() == 0)      //没有时就阻塞
        while (_q.size() == 0) // 防止伪唤醒的情况发生
            pthread_cond_wait(&_c_cond, &_mutex);
        T ret = _q.front();
        _q.pop();
        // if(_q.size() < _low_water)      //太少了，唤醒生产
        //     pthread_cond_signal(&_p_cond);
        pthread_cond_signal(&_p_cond);
        pthread_mutex_unlock(&_mutex);
        return ret;
    }

    void Push(const T &in) // 不会修改，可以加const修饰和用引用
    {
        pthread_mutex_lock(&_mutex);
        // if(_q.size() >= _max_cap)      //达到最大时就阻塞
        while (_q.size() >= _max_cap)
            pthread_cond_wait(&_p_cond, &_mutex); // 不要忘记加&
        _q.push(in);
        // if(_q.size() > _high_water)      //太多了，唤醒消费
        //     pthread_cond_signal(&_c_cond);
        pthread_cond_signal(&_c_cond);
        pthread_mutex_unlock(&_mutex);
    }

    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_c_cond);
        pthread_cond_destroy(&_p_cond);
        std::cout << "mutex and cond destroy" << std::endl;
    }

private:
    std::queue<T> _q;
    int _max_cap;
    pthread_mutex_t _mutex;
    pthread_cond_t _c_cond;
    pthread_cond_t _p_cond;
    // int _low_water;
    // int _high_water;
};